package org.example.env_boot.repository;

import org.example.env_boot.entity.Statistics;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.jpa.repository.Query;
import org.springframework.data.repository.query.Param;
import org.springframework.stereotype.Repository;

import java.util.List;

/**
 * 统计数据访问接口
 */
@Repository
public interface StatisticsRepository extends JpaRepository<Statistics, Integer> {
    
    /**
     * 根据省ID查找统计数据
     */
    List<Statistics> findByProvinceId(Integer provinceId);
    
    /**
     * 根据城市ID查找统计数据
     */
    List<Statistics> findByCityId(Integer cityId);
    
    /**
     * 根据省ID和城市ID查找统计数据
     */
    List<Statistics> findByProvinceIdAndCityId(Integer provinceId, Integer cityId);
    
    /**
     * 根据网格员ID查找统计数据
     */
    List<Statistics> findByGmId(Integer gmId);
    
    /**
     * 根据AQI级别查找统计数据
     */
    List<Statistics> findByAqiId(Integer aqiId);
    
    /**
     * 根据反馈ID查找统计数据
     */
    List<Statistics> findByFeedbackId(Integer feedbackId);
    
    /**
     * 根据统计日期查找统计数据
     */
    List<Statistics> findByDateTime(String dateTime);
    
    /**
     * 根据日期范围查找统计数据
     */
    @Query("SELECT s FROM Statistics s WHERE s.dateTime BETWEEN :startDate AND :endDate ORDER BY s.dateTime DESC")
    List<Statistics> findByDateRange(@Param("startDate") String startDate, @Param("endDate") String endDate);
    
    /**
     * 统计某个省的数据总数
     */
    @Query("SELECT COUNT(s) FROM Statistics s WHERE s.provinceId = :provinceId")
    Long countByProvinceId(@Param("provinceId") Integer provinceId);
    
    /**
     * 统计某个城市的数据总数
     */
    @Query("SELECT COUNT(s) FROM Statistics s WHERE s.cityId = :cityId")
    Long countByCityId(@Param("cityId") Integer cityId);
    
    /**
     * 统计某个AQI级别的数据总数
     */
    @Query("SELECT COUNT(s) FROM Statistics s WHERE s.aqiId = :aqiId")
    Long countByAqiId(@Param("aqiId") Integer aqiId);
    
    /**
     * 按省份统计各AQI级别的数量
     */
    @Query("SELECT s.provinceId, s.aqiId, COUNT(s) FROM Statistics s GROUP BY s.provinceId, s.aqiId")
    List<Object[]> countByProvinceAndAqiLevel();
    
    /**
     * 按城市统计各AQI级别的数量
     */
    @Query("SELECT s.cityId, s.aqiId, COUNT(s) FROM Statistics s GROUP BY s.cityId, s.aqiId")
    List<Object[]> countByCityAndAqiLevel();
    
    /**
     * 按月份统计数据数量
     */
    @Query("SELECT SUBSTRING(s.dateTime, 1, 7), COUNT(s) FROM Statistics s GROUP BY SUBSTRING(s.dateTime, 1, 7) ORDER BY SUBSTRING(s.dateTime, 1, 7)")
    List<Object[]> countByMonth();
} 